#ifndef DATA
#define DATA

#include <torch/torch.h>
torch::Tensor dataSet = torch::tensor({
            {1.0, 4.0}, 
            {2.0, 7.0}, 
            {3.0, 10.0}, 
            {4.0, 13.0}, 
            {5.0, 16.0},
            {6.0, 19.0},
            {7.0, 22.0},
            {8.0, 25.0},
            {9.0, 28.0},
            {10.0,31.0},
            {11.0,34.0},
            {12.0,37.0},
            {13.0,40.0},
            {14.0,43.0},
            {15.0,46.0}},
            torch::requires_grad(true));
#endif

// tensor的遍历:
// cout << dataSet[0] << endl;
// 输出: 1 4
//
// torch::Tensor array = dataSet[0];
// cout << "array:\n" << array << endl;
// cout << "array size:\n" << array.sizes() << endl;
// torch::Tensor a = torch::tensor({1,4});
// cout << "a size:\n" << a.sizes() << endl;
// 两者的size是相同的
